package com.learn.acm.one;

import org.junit.jupiter.api.Test;

/**
 * 给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
 *
 * 你可以假设数组中无重复元素。
 *
 * 示例 1:
 *
 * 输入: [1,3,5,6], 5
 * 输出: 2
 * 示例 2:
 *
 * 输入: [1,3,5,6], 2
 * 输出: 1
 * 示例 3:
 *
 * 输入: [1,3,5,6], 7
 * 输出: 4
 * 示例 4:
 *
 * 输入: [1,3,5,6], 0
 * 输出: 0
 */
public class Solution35 {

    @Test
    public void test35() {
        System.out.println(searchInsert(new int[]{1,3,5,6},2));
    }

    public int searchInsert(int[] nums, int target) {
        if (null == nums || target < nums[0]) return 0;
        if (nums[nums.length-1] < target) return nums.length;
        int start = 0,end = nums.length -1,mid = 0;
        int ans = 0;
        while (start <= end) {
            mid = start + (end - start) /2;
            if (target <= nums[mid]) {
                ans = mid;
                end = mid - 1;
            } else {
                start = mid + 1;
            }
        }
        return ans;
    }

    public int searchInsert1(int[] nums, int target) {
        if (null == nums || target < nums[0]) return 0;
        if (nums[nums.length-1] < target) return nums.length;
        for (int i = 0;i<nums.length;i++) {
            if (nums[i] == target) {
                return i;
            }
            if (nums[i+1] > target) {
                return i+1;
            }
        }
        return 0;
    }
}
